from modeltranslation.translator import translator, TranslationOptions
from photologue.models import Gallery, Photo


class GalleryTranslationOptions(TranslationOptions):
    fields = ('title', 'description',)

class PhotoTranslationOptions(TranslationOptions):
    fields = ('title', 'caption',)


translator.register(Gallery, GalleryTranslationOptions)
translator.register(Photo, PhotoTranslationOptions)